import socket

def send_message(udp_socket):
    message = input('请输入你想发送的信息：')
    IP = input('请输入发送的IP地址：')
    potr = int(input('请输入你要发送的端口号'))
    if potr:
        udp_socket.sendto(message.encode('utf-8'),(IP,potr))

def recv_message(udp_socket):
    message = udp_socket.recvfrom(1024)
    ip = message[0].decode('gbk')
    mes = message[1]
    print('%s:%s' % (str(ip),mes))

def main():
    #创建
    udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    #绑定 udp_socket 的端口号
    udp_socket.bind(('',8081))
    print('输入1：发送')
    print('输入2：接收')
    print('——' * 20)
    while True: #循环
        print('——'*20)
        op_mun = input('请输入你选择的功能序号：')
        if op_mun == '1':
            send_message(udp_socket)
        elif op_mun == '2':
            recv_message(udp_socket)
        else:
            print('输入有误')


if __name__ == '__main__':
    main()
